
#include "ednn.h"


struct ednn_layer *ednn_relu_layer(void)
{
    return EDNN_NULL;
}

static ednn_ret softmax_build(struct ednn_layer *layer)
{
    struct ednn_layer *prev_layer = ednn_list_entry(layer->list.prev, 
        struct ednn_layer, list);
    layer->output = ednn_tensor_create(prev_layer->output->dim, 
        prev_layer->output->ndim);
    return ednn_ok;
}

static ednn_ret softmax_forward(struct ednn_layer *layer)
{
    return ednn_ok;
}

struct ednn_layer *ednn_softmax_layer(void)
{
    struct ednn_layer *layer = ednn_mem_zalloc(sizeof(struct ednn_layer));
    ednn_assert(layer != EDNN_NULL);

    layer->type = EDNN_LAYER_SOFTMAX;
    layer->build = softmax_build;
    layer->forward = softmax_forward;

    return layer;
}